str1=input("输入上传的文件")

print(str1.endswith(".png"))   ##判断是否为png结尾的文件

if str1.endswith(".png") or str1.endswith(".jpg"):
    print("上传成功！")
else:
    print("非png或jpg文件，上传失败！")
